from game import zz_game
from zz import zz_log

log = zz_log.of()


# 我就爱消除

class Game(zz_game.ZZ_Game):
    def __init__(self, d):
        super().__init__(d)

    def get_name(self):
        return "冒险大作战"

    def game_alert(self):
        self.alert_close_by_file('./game/img/冒险大作战')

    def play_alert(self):
        if not self.img_exit("./game/img/冒险大作战/play_alert.png", 2):
            self.log_msg("不存在弹窗")
            return False

        self.log_msg("存在武器弹窗")
        if self.img_exit("./game/img/冒险大作战/play_state.png", 2):
            self.log_msg("战力下降，点击出售")
            self.img_click("./game/img/冒险大作战/play_buy.png")
        else:
            self.log_msg("战力上升，点击收服")
            self.img_click("./game/img/冒险大作战/play_get.png")

        if self.img_exit("./game/img/冒险大作战/play_confirm_alert.png", 1):
            self.log_msg("存在弹窗确认，点击确认")
            self.img_click("./game/img/冒险大作战/play_confirm_1.png", 1)
        return True

    def handle_main(self):
        self.log_msg("等待进行游戏主界面")
        self.img_exit('./game/img/冒险大作战/game_main.png', 100)
        self.log_msg("已进行游戏主界面")
        self.log_msg("检查是否存在弹窗")

        self.game_alert()

        if self.img_exit('./game/img/冒险大作战/get_prize.jpg', 3):
            self.log_msg("存在奖励弹窗，进行领取")
            self.img_click('./game/img/冒险大作战/get_prize.jpg')
            self.click_bottom()

        self.log_msg("点击自动吹灯")
        self.img_click('./game/img/冒险大作战/auto.jpg')
        if self.img_exit('./game/img/冒险大作战/all_buy.jpg', 3):
            self.log_msg("存在需要出售的准备，进行出售")
            self.img_click('./game/img/冒险大作战/all_buy.jpg')
            if self.img_exit("./game/img/冒险大作战/play_confirm_alert.png", 1):
                self.log_msg("存在确认弹窗，点击确定")
                self.img_click('./game/img/冒险大作战/play_buy_confirm_1.png')

            self.game_alert()
            self.img_click_if_exists("点击自动吹灯", './game/img/冒险大作战/auto.jpg', './game/img/冒险大作战/start.jpg')

        self.img_click_for('点击开始', './game/img/冒险大作战/start.jpg')
        while True:
            self.game_alert()
            self.play_alert()

            self.log_msg("检查任务是否完成")
            if self.check_done():
                self.log_msg("任务已完成")
                break
            self.d.sleep(3)

        self.d.sleep(3)
        return True
